from init_sql import init_sql
from functions import *



if __name__ == '__main__':
    while True:
        print('''
    请输入你的选择：
    1. 信息录入
    2. 查询信息
    3. 删除信息
    4. 修改工程零件数
    5. 浏览数据库
    6. 初始化数据库
    0. 退出
        ''')
        print('> ', end='')
        choice = input()
        if choice == '1':
            insert()
        elif choice == '2':
            search()
        elif choice == '3':
            delete()
        elif choice == '4':
            update()
        elif choice == '5':
            watch()
        elif choice == '0':
            break
        elif choice == '6':
            init_sql()
        else:
            print('输入错误，请重新输入！')